# -*- coding: utf-8 -*-
# @Time    : 2024/5/13 09:56
# @Author  : Hean
# @Site    : 
# @File    : config.py
# @Software: PyCharm


from pydantic_settings import BaseSettings


class Settings(BaseSettings):
    dify_url: str
    dify_api_key: str
    queue_push_sever: str
    getdata_topic_name: str = "hean_liveroom_getdata_topic"
    generate_reply_topic: str = "hean_liveroom_generate_reply_topic"
    group_id: str = 'Generate_reply_test520'
    database_url: str
    database_port: int
    database_db: str
    database_user: str
    database_password: str
    max_workers: int = 2  # 生成回复的最大线程数，可以根据需要调整线程数
    max_time_interval: int = 15  # 超过15秒的不处理
    reply_indicate_user: bool = False  # 回复的消息中@用户和不@用户（看直播间是否是匿名）

    class Config:
        env_file = ".env"
        env_file_encoding = "utf-8"


settings = Settings()
